package vn.edu.hut.soict.translate.wordgraph;

import java.util.Collections;
import java.util.List;

import vn.edu.hut.soict.automaton.DFA;
import vn.edu.hut.soict.automaton.State;

public class WordGraph extends DFA<Word> {
	public static final String GRAPH_FILE = "res/graph.xml";

	private int contentWordCount;
	private int functionWordCount;

	public WordGraph() {

	}

	public void reindex() {
		List<State<Word>> states = getStates();
		states.removeAll(Collections.singletonList(null));
		for (State<Word> state : states) {
			state.setId(states.indexOf(state));
		}
	}

	public int getContentWordCount() {
		return contentWordCount;
	}

	public void setContentWordCount(int contentWordCount) {
		this.contentWordCount = contentWordCount;
	}

	public int getFunctionWordCount() {
		return functionWordCount;
	}

	public void setFunctionWordCount(int functionWordCount) {
		this.functionWordCount = functionWordCount;
	}
}
